package com.grow.greenhouse.utils;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTVerifier;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.JWTVerificationException;
import com.auth0.jwt.interfaces.DecodedJWT;
import java.util.Date;

public class JwtUtil {
    private static final String SECRET = "GREEN_HOUSE"; // 密钥，应该保密且复杂
    private static final long EXPIRE_TIME = 5 * 60 * 60 * 1000; // 设置token过期时间，例如这里设置为5小时

    /**
     * 生成JWT token
     *
     * @param userId 用户信息（目前包含用户id、用户当前选择的棚名称）
     * @return 返回生成的token
     */
    public static String createToken(Long userId) {
        Date now = new Date();
        Algorithm algorithm = Algorithm.HMAC256(SECRET);
        return JWT.create()
                .withSubject(userId.toString()) // 将userInfo保存到token里作为载荷
                .withIssuedAt(now) // token的创建时间
                .withExpiresAt(new Date(now.getTime() + EXPIRE_TIME)) // 设置token过期时间
                .sign(algorithm); // 使用HMAC256算法，以及密钥签名token
    }

    /**
     * 验证token的有效性
     *
     * @param token 要验证的token
     * @return 如果token有效，返回true；否则返回false
     */
    public static boolean verifyToken(String token) {
        try {
            Algorithm algorithm = Algorithm.HMAC256(SECRET); // 使用HMAC256算法
            JWTVerifier verifier = JWT.require(algorithm).build(); // 创建token验证器
            DecodedJWT jwt = verifier.verify(token); // 验证token
            return true; // 验证通过，返回true
        } catch (JWTVerificationException exception) {
            // token验证失败，返回false
            return false;
        }
    }

    /**
     * 从token中获取用户信息
     *
     * @param token token
     * @return 返回从token中解析出的用户id
     */
    public static String getUserIdFromToken(String token) {
        DecodedJWT jwt = JWT.decode(token);
        return jwt.getSubject();
    }
}
